home *** CD-ROM | disk | FTP | other *** search
/ Atari Mega Archive 1 / Atari Mega Archive - Volume 1.iso / telecomm / sticpsrc.lzh / SOURCE.ARC / TELNET.C < prev    next >
C/C++ Source or Header  |  1990-02-04  |  10KB  |  484 lines
  1. #include <stdio.h>
  2. #include "global.h"
  3. #include "mbuf.h"
  4. #include "timer.h"
  5. #include "icmp.h"
  6. #include "netuser.h"
  7. #include "tcp.h"
  8. #include "telnet.h"
  9. #include "session.h"
  10.  
  11. #define CTLZ    26
  12.  
  13. extern char nospace[];
  14. extern char badhost[];
  15. int refuse_echo = 0;
  16. int unix_line_mode = 0;       /* if true turn <cr> to <nl> when in line mode */
  17.  
  18. #ifdef    DEBUG
  19. char *t_options[] = {
  20.     "Transmit Binary",
  21.     "Echo",
  22.     "",
  23.     "Suppress Go Ahead",
  24.     "",
  25.     "Status",
  26.     "Timing Mark"
  27. };
  28. #endif
  29.  
  30. /* Execute user telnet command */
  31. int
  32. dotelnet(argc,argv)
  33. int argc;
  34. char *argv[];
  35. {
  36.     void t_state(),rcv_char(),tn_tx();
  37.     int32 resolve();
  38.     int send_tel();
  39.     int unix_send_tel();
  40.     struct session *s;
  41.     struct telnet *tn;
  42.     struct tcb *tcb;
  43.     struct socket lsocket,fsocket;
  44.     char tos = 0;
  45.  
  46.     lsocket.address = ip_addr;
  47.     lsocket.port = lport++;
  48.     if((fsocket.address = resolve(argv[1])) == 0){
  49.         printf(badhost,argv[1]);
  50.         return 1;
  51.     }
  52.     if(argc < 3)
  53.         fsocket.port = TELNET_PORT;
  54.     else {
  55.         if((fsocket.port = atoi(argv[2])) == 0)
  56.             fsocket.port = TELNET_PORT;
  57.         tos = get_tos(argv[3]);
  58.     }
  59.     /* Allocate a session descriptor */
  60.     if((s = newsession()) == NULLSESSION){
  61.         printf("Too many sessions\n");
  62.         return 1;
  63.     }
  64.     if((s->name = malloc((unsigned)strlen(argv[1])+1)) != NULLCHAR)
  65.         strcpy(s->name,argv[1]);
  66.     s->type = TELNET;
  67.     if ((refuse_echo == 0) && (unix_line_mode != 0)) {
  68.         s->parse = unix_send_tel;
  69.     } else {
  70.         s->parse = send_tel;
  71.     }
  72.     current = s;
  73.  
  74.     /* Create and initialize a Telnet protocol descriptor */
  75.     if((tn = (struct telnet *)calloc(1,sizeof(struct telnet))) == NULLTN){
  76.         printf(nospace);
  77.         s->type = FREE;
  78.         return 1;
  79.     }
  80.     tn->session = s;    /* Upward pointer */
  81.     tn->state = TS_DATA;
  82.     s->cb.telnet = tn;    /* Downward pointer */
  83.  
  84.     tcb = open_tcp(&lsocket,&fsocket,TCP_ACTIVE,0,
  85.      rcv_char,tn_tx,t_state,tos,(char *)tn);
  86.  
  87.     tn->tcb = tcb;    /* Downward pointer */
  88.     go();
  89.     return 0;
  90. }
  91.  
  92. /* Process typed characters */
  93. int
  94. unix_send_tel(buf,n)
  95. char *buf;
  96. int16 n;
  97. {
  98.     int i;
  99.  
  100.     for (i=0; (i<n) && (buf[i] != '\r'); i++)
  101.         ;
  102.     if (buf[i] == '\r') {
  103.         buf[i] = '\n';
  104.         n = i+1;
  105.     }
  106.     send_tel(buf,n);
  107. }
  108. int
  109. send_tel(buf,n)
  110. char *buf;
  111. int16 n;
  112. {
  113.     struct mbuf *bp,*qdata();
  114.     if(current == NULLSESSION || current->cb.telnet == NULLTN
  115.      || current->cb.telnet->tcb == NULLTCB)
  116.         return;
  117.     /* If we're doing our own echoing and recording is enabled, record it */
  118.     if(!current->cb.telnet->remote[TN_ECHO] && current->record != NULLFILE)
  119.         fwrite(buf,1,n,current->record);
  120.     bp = qdata(buf,n);
  121.     send_tcp(current->cb.telnet->tcb,bp);
  122. }
  123.  
  124. /* Process incoming TELNET characters */
  125. int
  126. tel_input(tn,bp)
  127. register struct telnet *tn;
  128. struct mbuf *bp;
  129. {
  130.     char c;
  131.     void doopt(),dontopt(),willopt(),wontopt(),answer();
  132.     FILE *record;
  133. #if (!defined(MSC) && !defined(__TURBOC__))
  134.     char *memchr();
  135. #endif
  136.  
  137.     /* Optimization for very common special case -- no special chars */
  138.     if(tn->state == TS_DATA){
  139.         while(bp != NULLBUF && memchr(bp->data,IAC,bp->cnt) == NULL){
  140.             if((record = tn->session->record) != NULLFILE)
  141.                 fwrite(bp->data,1,bp->cnt,record);
  142.             while(bp->cnt-- != 0)
  143.                 putchar(*bp->data++);
  144.             bp = free_mbuf(bp);
  145.         }
  146.     }
  147.     while(pullup(&bp,&c,1) == 1){
  148.         switch(tn->state){
  149.         case TS_DATA:
  150.             if(uchar(c) == IAC){
  151.                 tn->state = TS_IAC;
  152.             } else {
  153.                 if(!tn->remote[TN_TRANSMIT_BINARY])
  154.                     c &= 0x7f;
  155.                 putchar(c);
  156.                 if((record = tn->session->record) != NULLFILE)
  157.                     putc(c,record);
  158.             }
  159.             break;
  160.         case TS_IAC:
  161.             switch(uchar(c)){
  162.             case WILL:
  163.                 tn->state = TS_WILL;
  164.                 break;
  165.             case WONT:
  166.                 tn->state = TS_WONT;
  167.                 break;
  168.             case DO:
  169.                 tn->state = TS_DO;
  170.                 break;
  171.             case DONT:
  172.                 tn->state = TS_DONT;
  173.                 break;
  174.             case IAC:
  175.                 putchar(c);
  176.                 tn->state = TS_DATA;
  177.                 break;
  178.             default:
  179.                 tn->state = TS_DATA;
  180.                 break;
  181.             }
  182.             break;
  183.         case TS_WILL:
  184.             willopt(tn,c);
  185.             tn->state = TS_DATA;
  186.             break;
  187.         case TS_WONT:
  188.             wontopt(tn,c);
  189.             tn->state = TS_DATA;
  190.             break;
  191.         case TS_DO:
  192.             doopt(tn,c);
  193.             tn->state = TS_DATA;
  194.             break;
  195.         case TS_DONT:
  196.             dontopt(tn,c);
  197.             tn->state = TS_DATA;
  198.             break;
  199.         }
  200.     }
  201. }
  202.  
  203. /* Telnet receiver upcall routine */
  204. void
  205. rcv_char(tcb,cnt)
  206. register struct tcb *tcb;
  207. int16 cnt;
  208. {
  209.     struct mbuf *bp;
  210.     struct telnet *tn;
  211.     FILE *record;
  212.     extern int flowctrl,ttydriv();
  213.  
  214.     if((tn = (struct telnet *)tcb->user) == NULLTN){
  215.         /* Unknown connection; ignore it */
  216.         return;
  217.     }
  218.     /* Hold output if we're not the current session, and while typing */
  219.     if(mode != CONV_MODE || current == NULLSESSION
  220.      || current->type != TELNET || current->cb.telnet != tn
  221.      || (flowctrl && ttydriv(-1,NULL)))
  222.         return;
  223.  
  224.     if(recv_tcp(tcb,&bp,cnt) > 0)
  225.         tel_input(tn,bp);
  226.  
  227.     fflush(stdout);
  228.     if((record = tn->session->record) != NULLFILE)
  229.         fflush(record);
  230. }
  231. /* Handle transmit upcalls. Used only for file uploading */
  232. void
  233. tn_tx(tcb,cnt)
  234. struct tcb *tcb;
  235. int16 cnt;
  236. {
  237.     struct telnet *tn;
  238.     struct session *s;
  239.     struct mbuf *bp;
  240.     int size;
  241.  
  242.     if((tn = (struct telnet *)tcb->user) == NULLTN
  243.      || (s = tn->session) == NULLSESSION
  244.      || s->upload == NULLFILE)
  245.         return;
  246.     while((bp = alloc_mbuf(cnt)) == NULLBUF && (cnt /= 2) != 0)
  247.         ;
  248.     if(bp == NULLBUF)
  249.         return;
  250.     if((size = fread(bp->data,1,cnt,s->upload)) > 0){
  251.         bp->cnt = (int16)size;
  252.         send_tcp(tcb,bp);
  253.     } else {
  254.         free_p(bp);
  255.     }
  256.     if(size != cnt){
  257.         /* Error or end-of-file */
  258.         fclose(s->upload);
  259.         s->upload = NULLFILE;
  260.         free(s->ufile);
  261.         s->ufile = NULLCHAR;
  262.     }
  263. }
  264.  
  265. /* State change upcall routine */
  266. void
  267. t_state(tcb,old,new)
  268. register struct tcb *tcb;
  269. char old,new;
  270. {
  271.     struct telnet *tn;
  272.     char notify = 0;
  273.     extern char *tcpstates[];
  274.     extern char *reasons[];
  275.     extern char *unreach[];
  276.     extern char *exceed[];
  277.  
  278.     /* Can't add a check for unknown connection here, it would loop
  279.      * on a close upcall! We're just careful later on.
  280.      */
  281.     tn = (struct telnet *)tcb->user;
  282.  
  283.     if(current != NULLSESSION && current->type == TELNET && current->cb.telnet == tn)
  284.         notify = 1;
  285.  
  286.     switch(new){
  287.     case CLOSE_WAIT:
  288.         if(notify)
  289.             printf("%s\n",tcpstates[new]);
  290.         close_tcp(tcb);
  291.         break;
  292.     case CLOSED:    /* court adjourned */
  293.         if(notify){
  294.             printf("%s (%s",tcpstates[new],reasons[tcb->reason]);
  295.             if(tcb->reason == NETWORK){
  296.                 switch(tcb->type){
  297.                 case DEST_UNREACH:
  298.                     printf(": %s unreachable",unreach[tcb->code]);
  299.                     break;
  300.                 case TIME_EXCEED:
  301.                     printf(": %s time exceeded",exceed[tcb->code]);
  302.                     break;
  303.                 }
  304.             }
  305.             printf(")\n");
  306.             cmdmode();
  307.         }
  308.         del_tcp(tcb);
  309.         if(tn != NULLTN)
  310.             free_telnet(tn);
  311.         break;
  312.     default:
  313.         if(notify)
  314.             printf("%s\n",tcpstates[new]);
  315.         break;
  316.     }
  317.     fflush(stdout);
  318. }
  319. /* Delete telnet structure */
  320. static
  321. free_telnet(tn)
  322. struct telnet *tn;
  323. {
  324.     if(tn->session != NULLSESSION)
  325.         freesession(tn->session);
  326.  
  327.     if(tn != NULLTN)
  328.         free((char *)tn);
  329. }
  330.  
  331. /* The guts of the actual Telnet protocol: negotiating options */
  332. static
  333. void
  334. willopt(tn,opt)
  335. struct telnet *tn;
  336. char opt;
  337. {
  338.     int ack;
  339.     void answer();
  340.  
  341. #ifdef    DEBUG
  342.     printf("recv: will ");
  343.     if(uchar(opt) <= NOPTIONS)
  344.         printf("%s\n",t_options[opt]);
  345.     else
  346.         printf("%u\n",opt);
  347. #endif
  348.  
  349.     switch(uchar(opt)){
  350.     case TN_TRANSMIT_BINARY:
  351.     case TN_ECHO:
  352.     case TN_SUPPRESS_GA:
  353.         if(tn->remote[uchar(opt)] == 1)
  354.             return;        /* Already set, ignore to prevent loop */
  355.         if(uchar(opt) == TN_ECHO){
  356.             if(refuse_echo){
  357.                 /* User doesn't want to accept */
  358.                 ack = DONT;
  359.                 break;
  360.             } else
  361.                 raw();        /* Put tty into raw mode */
  362.         }
  363.         tn->remote[uchar(opt)] = 1;
  364.         ack = DO;
  365.         break;
  366.     default:
  367.         ack = DONT;    /* We don't know what he's offering; refuse */
  368.     }
  369.     answer(tn,ack,opt);
  370. }
  371. static
  372. void
  373. wontopt(tn,opt)
  374. struct telnet *tn;
  375. char opt;
  376. {
  377.     void answer();
  378.  
  379. #ifdef    DEBUG
  380.     printf("recv: wont ");
  381.     if(uchar(opt) <= NOPTIONS)
  382.         printf("%s\n",t_options[uchar(opt)]);
  383.     else
  384.         printf("%u\n",uchar(opt));
  385. #endif
  386.     if(uchar(opt) <= NOPTIONS){
  387.         if(tn->remote[uchar(opt)] == 0)
  388.             return;        /* Already clear, ignore to prevent loop */
  389.         tn->remote[uchar(opt)] = 0;
  390.         if(uchar(opt) == TN_ECHO)
  391.             cooked();